package io.gitee.twoke.generator.engine.impl.generate.parameter;

import io.gitee.twoke.generator.engine.impl.generate.DataTable;
import io.gitee.twoke.generator.engine.impl.generate.processor.GenerateExecutorImpl;


public class EntityParameterLoader implements ParameterLoader{

    @Override
    public void load(ParameterContext context) {
        DataTable dataTable = (DataTable)context.get(GenerateExecutorImpl.DATA);
        String className = dataTable.getClassName();
        context.put("className", className);
    }

    @Override
    public void unload(ParameterContext context) {
        context.remove("className");
    }

}
